package com.util;

import java.io.*;

public class FileUtil {
    /**
     * 从资源目录读取文件并返回byte数组。
     *
     * @param resourcePath 文件在资源目录下的路径。
     * @return 文件的byte数组。
     * @throws IOException 如果文件读取过程中发生错误。
     */
    public static byte[] readFileAsByteArray(String resourcePath) throws IOException {
        // 使用ClassLoader获取资源输入流
        InputStream inputStream = FileUtil.class.getClassLoader().getResourceAsStream(resourcePath);
        if (inputStream == null) {
            throw new IOException("Resource not found: " + resourcePath);
        }

        // 读取所有字节
        byte[] buffer = new byte[8192];
        int bytesRead;
        ByteArrayOutputStream output = new ByteArrayOutputStream();
        try {
            while ((bytesRead = inputStream.read(buffer)) != -1) {
                output.write(buffer, 0, bytesRead);
            }
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            try {
                output.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }

        return output.toByteArray();
    }

    /**
     * 将byte转换为一个文件
     *
     * @param sourceBytes 源 byte数组
     * @param targetFile  目标文件
     * @return 返回是否成功转换
     */
    public static boolean byteToFile(byte[] sourceBytes, File targetFile) {
        if (sourceBytes == null || targetFile == null) {
            return false;
        }

        try (FileOutputStream fos = new FileOutputStream(targetFile)) {
            fos.write(sourceBytes);
            return true; // 写入成功
        } catch (IOException e) {
            e.printStackTrace();
            return false; // 发生异常，写入失败
        }
    }

    /**
     * 将一个文件转换为 byte数组
     */
    public static byte[] fileToByte(File sourceFile) {
        // 使用 byte[] 而不是 Byte[]，因为 byte[] 是基本类型数组，更高效
        byte[] result = null;
        FileInputStream fis = null;
        try {
            // 检查文件是否存在
            if (!sourceFile.exists()) {
                return null;
            }

            // 获取文件长度
            long length = sourceFile.length();
            // 如果文件长度超过 Integer.MAX_VALUE，则返回 null
            if (length > Integer.MAX_VALUE) {
                return null;
            }

            // 创建 FileInputStream 对象
            fis = new FileInputStream(sourceFile);
            // 创建 byte 数组以保存文件数据
            result = new byte[(int) length];

            // 读取文件内容到 byte 数组中
            int offset = 0;
            int numRead = 0;
            while (offset < result.length && (numRead = fis.read(result, offset, result.length - offset)) >= 0) {
                offset += numRead;
            }

            // 确保所有数据都被读取
            if (offset < result.length) {
                throw new IOException("Could not completely read file " + sourceFile.getName());
            }
        } catch (IOException e) {
            // 打印异常信息并返回 null
            e.printStackTrace();
            return null;
        } finally {
            // 关闭 FileInputStream
            if (fis != null) {
                try {
                    fis.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }

        return result;
    }
}
